import torch
import torch.version

print("Hello World, Hello Pytorch {}".format(torch.version))
print("\nCUDA is available:{}, version is {}".format(torch.cuda.is_available(), torch.version.cuda))
print("\ndevice_name:{}".format(torch.cuda.get_device_name(0)))

x = torch.rand(5,3)
print(x)